#!/usr/bin/python3
# -*- coding:utf-8 -*-
""""
@Author: 李吉凯
@Email: lijk34925@hundsun.com
@Time: 2022/5/23 14:01
@File: 风险等级查询_导出.py
"""
from qturboframe.lib.utils.by import By

from BladeFerm.Lib.Common import logger, data_processing
from BladeFerm.hui12_common.check_data import check_data_not_contains, check_file_exit
from BladeFerm.hui2_app import FE
from BladeFerm.Lib.Common import DownPath
from BladeFerm.utils.basic_opration import clear_dir
from 基础公共脚本.公共方法_hui2 import open_menu




@data_processing
def customer_rate_process_export_data(input_data=None, ui_expect=None):
    logger.debug(f'DownPath： {DownPath}')
    # 进入 风险等级查询 菜单
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '客户风险评估/客户评级流程查询'})

    # 清空目录
    clear_dir(DownPath)

    # 导出 数据
    grid = FE.table_new({'text': '唯一客户号'})
    is_all = True if input_data.get('导出全部否', '是') == '是' else False
    grid.export(is_all=is_all, rows=input_data.get('导出数据行', None))

    # 校验
    check_file_exit('客户评级流程查询', timeout=int(input_data.get('超时时间', 10)))

    # # 检查导出结果
    # box = FE.msg_box(idname={By.XPATH: '//div[@class="h-modal-wrap h-modal-nocopy"]'}, timeout=1)
    # box.check_dispaly(ui_expect.get('期望结果') == '是')
    # # 等待导出弹框关闭
    # if box.element:
    #     box.check_export_result(True)

def customer_rate_process_refresh_data(input_data=None, ui_expect=None):
    # 进入 风险等级查询 菜单
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '客户风险评估/客户评级流程查询'})
    grid = FE.table_new({'text': '唯一客户号'})
    grid.refresh()

@data_processing
def customer_rate_process_set_col_name(input_data=None, ui_expect=None):
    logger.debug(f'DownPath： {DownPath}')
    # 进入 风险等级查询 菜单
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '客户风险评估/客户评级流程查询'})
    grid = FE.table_new()
    grid.refresh()
    cancel_cols = input_data.get('取消列名')
    grid.set_col_name(cancel_cols)
    col_name = grid.get_title_list()
    check_data_not_contains(cancel_cols, col_name)
